www.gusucode.com > Deep Sudoku Solver工具箱matlab源码程序 > Deep Sudoku Solver/src/sudoku/+sudoku/+training/resnetLike.m
function lgraph = resnetLike() % Copyright 2018 The MathWorks, Inc. net = resnet50(); lgraph = layerGraph(net); lgraph = removeLayers(lgraph, ... {'input_1', ... 'avg_pool', ... 'fc1000', ... 'fc1000_softmax', ... 'ClassificationLayer_fc1000'}); lgraph = addLayers(lgraph, imageInputLayer([64, 64, 3], 'Name', 'input')); lgraph = connectLayers(lgraph, 'input', 'conv1'); numClasses = 10; newLayers = [ averagePooling2dLayer([2, 2], 'Name', 'avg_pool') fullyConnectedLayer(numClasses,'Name','fc') softmaxLayer('Name','softmax') classificationLayer('Name','classoutput')]; lgraph = addLayers(lgraph,newLayers); lgraph = connectLayers(lgraph,'activation_49_relu','avg_pool'); end